package idv.arthur.work;

import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.google.gson.Gson;

/**
 * Servlet implementation class DelMRitem
 */ 
public class QueryPersonnel extends HttpServlet {
	private static final long serialVersionUID = 1L;
       
    public QueryPersonnel() {
        super();
        // TODO Auto-generated constructor stub
    }
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    	doPost(request,response);
    }

	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		
		request.setCharacterEncoding("UTF-8");
		
		Widgets wid = new Widgets();
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		String strSQL = "SELECT COUNT(userid) FROM personnel";
		String dataType = request.getParameter("getDataType");
		ArrayList<InfoStruc> infoStrucs = new ArrayList<InfoStruc>();
		
		ResultTableSet resultTableSet = new ResultTableSet();
		
		int page = Integer.parseInt(request.getParameter("page"));
		int total_pages = 1;
		int count  = 0;
		int limit = Integer.parseInt(request.getParameter("rows"));
		
		if ("JSON".equalsIgnoreCase(dataType)) {
			strSQL = "SELECT userid,username FROM personnel ORDER BY userid ASC";
			try {
				conn = wid.getConn(true);
				ps = conn.prepareStatement(strSQL);
				rs = ps.executeQuery();
				while (rs.next()) {
					InfoStruc infoStruc = new InfoStruc();
					infoStruc.setId(rs.getString("userid"));
					infoStruc.setName(rs.getString("username"));
					infoStrucs.add(infoStruc);
				}	
			} catch ( SQLException e) {
				System.out.println(wid.errProcess(e));
			} 
		} else {
			try {
				conn = wid.getConn(true);	
				ps = conn.prepareStatement(strSQL);
				rs = ps.executeQuery();
				rs.next();
				count  = rs.getInt(1);
				
				if ( count  != 0 ) {
					
					if (limit  < count) {
						total_pages = (count / limit)+1;	
					}
					
					strSQL = "SELECT userid,username,email FROM personnel ORDER BY userid ASC";
					ps = conn.prepareStatement(strSQL);
					rs = ps.executeQuery();
					ArrayList<Rows> rows = new ArrayList<Rows>();
					
					while ( rs.next() ) {
						String userid = rs.getString("userid"); 
						Rows row = new Rows();
						row.setId(userid);
						String[] cell = new String[3];
						cell[0] = userid;
						cell[1] = rs.getString("username");
						cell[2] = rs.getString("email");
						row.setCell(cell);
						rows.add(row);
					}
					resultTableSet.setRows(rows);
					
				} else {
					page =0;
					total_pages=0;
				}
				
				resultTableSet.setPage(String.valueOf(page));
				resultTableSet.setTotal(total_pages);
				resultTableSet.setRecords(String.valueOf(count));
				
			} catch ( SQLException e) {
				System.out.println(wid.errProcess(e));
			}
		}
		
		try {
			if (rs != null) {rs.close();}
			if (ps!= null) {ps.close();}
			if (conn!= null) {conn.close();}
		} catch (SQLException e) {
			System.out.println(wid.errProcess(e));
		}
		
		
		
		response.setContentType("application/json");
		response.setCharacterEncoding("UTF-8");
		if ("JSON".equalsIgnoreCase(dataType)) {
			response.getWriter().write(new Gson().toJson(infoStrucs));
		} else {
			response.getWriter().write(new Gson().toJson(resultTableSet));	
		}
				
	}
}